package com.sky;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

/**
 * @author: 尹志伟
 * @date: 2023-01-10 13:57
 * @Description TODO
 **/
public class MpCodeCreate {
    public static void main(String[] args) {
        //1. 创建代码生成器对象，执行生成代码操作
        AutoGenerator autoGenerator = new AutoGenerator();

        //@TODO 设置数据源连接配置
        DataSourceConfig dataSource = new DataSourceConfig();
        dataSource.setDriverName("com.mysql.cj.jdbc.Driver");
        dataSource.setUrl("jdbc:mysql://localhost:3306/sky_take_out?serverTimezone=UTC");
        dataSource.setUsername("root");
        dataSource.setPassword("root");
        dataSource.setDbType(DbType.MYSQL);
        autoGenerator.setDataSource(dataSource);

        //@TODO 全局设置
        GlobalConfig globalConfig = new GlobalConfig();
        //设置生成目录
        globalConfig.setOutputDir("D:/page_2022/sky-take-out/sky-create/src/main/java");
        //设置生成完毕后是否打开生成代码所在的目录
        globalConfig.setOpen(true);
        //设置作者
        globalConfig.setAuthor("尹志伟");
        //设置是否覆盖原始生成的文件
        globalConfig.setFileOverride(true);
        //数据库 Mapper名称
        globalConfig.setMapperName("%sDao");
        //数据库 service名称
        globalConfig.setServiceName("%sService");
        //数据库 Controller名称
        globalConfig.setControllerName("%sController");
        //设置Id生成策略
        globalConfig.setIdType(IdType.AUTO);
        //开启Swagger2模式
        globalConfig.setSwagger2(true);

        autoGenerator.setGlobalConfig(globalConfig);


        //@TODO 包名设置
        PackageConfig packageInfo = new PackageConfig();
        //设置生成的包名，与代码所在位置不冲突，二者叠加组成完
        packageInfo.setParent("com.sky");
        //设置实体类包名
        packageInfo.setEntity("po");
        //设置数据层包名
        packageInfo.setMapper("mapper");
        autoGenerator.setPackageInfo(packageInfo);

        //@TODO 策略设置
        StrategyConfig strategyConfig = new StrategyConfig();
        //设置当前参与生成的表名，参数为可变参数
        strategyConfig.setInclude("address_book","category","dish","dish_flavor","employee", "order_detail","orders","setmeal","setmeal_dish","shopping_cart","user");

        //设置数据库表的前缀名称，模块名 = 数据库表名 - 前缀名  例如： User = tbl_user - tbl_
        strategyConfig.setTablePrefix("tb_");

        //设置是否启用Rest风格
        strategyConfig.setRestControllerStyle(true);
        //设置乐观锁字段名
        strategyConfig.setVersionFieldName("version");
        //设置逻辑删除字段名
        strategyConfig.setLogicDeleteFieldName("is_deleted");
        //设置  链式模型
        strategyConfig.setChainModel(true);
        //类名生成遵循驼峰命名
        strategyConfig.setNaming(NamingStrategy.underline_to_camel);
        //字段生成遵循驼峰命
        strategyConfig.setColumnNaming(NamingStrategy.underline_to_camel);
        //设置是否启用lombok
        strategyConfig.setEntityLombokModel(true);
        autoGenerator.setStrategy(strategyConfig);

        //@TODO 执行生成操作
        autoGenerator.execute();
    }
}
